package com.example.lcpractice.heading_offer;

import com.example.lcpractice.datastructure.ListNode;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * [剑指 Offer 22]链表中倒数第k个节点.java
 * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
 * <p>
 * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
 * <p>
 * 示例：
 * <p>
 * 给定一个链表: 1->2->3->4->5, 和 k = 2.
 * <p>
 * 返回链表 4->5.
 * Related Topics
 * 链表
 * 双指针
 */
public class q22 {

    public static void main(String[] args) {
        int[] nums = {1, 8, 9, 6, 4, 3, 7, 22};
        ListNode singleLinkedList2 = ListNode.getSingleLinkedList2();
        ListNode listNode1 = new ListNode(1);
        ListNode listNode2 = new ListNode(2);
        ListNode listNode3 = new ListNode(3);
        ListNode listNode4 = new ListNode(4);
        ListNode listNode5 = new ListNode(5);
        listNode1.setNext(listNode2);
        listNode2.setNext(listNode3);
        listNode3.setNext(listNode4);
        listNode4.setNext(listNode5);
//        singleLinkedList2.next = null;
        System.out.println(getKthFromEnd2(listNode1, 5));
    }


    public static ListNode getKthFromEnd(ListNode head, int k) {
        int length = getListLength(head);
        int index = 0;
        while (head != null) {
            if (k == length - index) {
                return head;
            }
            index++;
            head = head.next;
        }
        return null;
    }

    public static int getListLength(ListNode node) {
        int count = 0;
        while (node != null) {
            count++;
            node = node.next;
        }
        return count;
    }

    /**
     * 快慢指针间隔为k
     * @param head 传入头指针
     * @param k 倒数第k个
     * @return 指定节点
     */
    public static ListNode getKthFromEnd2(ListNode head, int k) {
        ListNode fast = head;
        ListNode slow = head;
        while (fast != null && k > 0) {
            fast = fast.next;
            k--;
        }
        while (fast != null) {
            fast = fast.next;
            slow = slow.next;
        }
        return slow;
    }
}
